package medium.string;

import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;

/**
 * <a href="https://leetcode-cn.com/problems/custom-sort-string">791. 自定义字符串排序</a>
 * 字符串S和 T 只包含小写字符。在S中，所有字符只会出现一次。
 * S 已经根据某种规则进行了排序。我们要根据S中的字符顺序对T进行排序。更具体地说，如果S中x在y之前出现，那么返回的字符串中x也应出现在y之前。
 * 返回任意一种符合条件的字符串T。
 * 示例:
 *   输入: S = "cba"，T = "abcd"
 *   输出: "cbad"
 *   解释: S中出现了字符 "a", "b", "c", 所以 "a", "b", "c" 的顺序应该是 "c", "b", "a".
 *   由于 "d" 没有在S中出现, 它可以放在T的任意位置. "dcba", "cdba", "cbda" 都是合法的输出。
 * 注意:
 *   S的最大长度为26，其中没有重复的字符。
 *   T的最大长度为200。
 *   S和T只包含小写字符。
 * 解题思路：
 *   1. 遍历t中的每个字符，将其转化为对应的ascii码，统计其出现次数，并按ascii值作为下标，将次数存入int数组中
 *   2. 定义StringBuilder对象，用来追加字符
 *   3. 遍历s中的每个字符，取其ascii值，作为下标定位到int数组中的某个元素，从而找到该字符在t中出现的次数
 *   4. 以字符出现的次数作为循环次数，每次将该字符追加到StringBuilder中，循环完后将次数清0
 *   5. 从字符a到字符z开始遍历，将t中出现的其他字符按步骤4的方式也追加到StringBuilder中
 *   6. 返回StringBuilder中的字符串
 * @author 刘学松
 * @date 2021/2/19 21:13
 */
public class 自定义字符串排序 {
    public static void main(String[] args) {
        String s = "cba";
        String t = "abcd";
        System.out.println(customSortString(s, t));
    }

    public static String customSortString(String s, String t) {
        int[] count = new int[26];
        for (char c: t.toCharArray()) {
            count[c - 'a']++;
        }

        StringBuilder ans = new StringBuilder();

        for (char c: s.toCharArray()) {
            for (int i = 0; i < count[c - 'a']; ++i) {
                ans.append(c);
            }
            count[c - 'a'] = 0;
        }

        for (char c = 'a'; c <= 'z'; ++c) {
            for (int i = 0; i < count[c - 'a']; ++i) {
                ans.append(c);
            }
        }

        return ans.toString();
    }

    public static String customSortString1(String s, String t) {
        Character[] chars = new Character[t.length()];
        Map<Character, Integer> map = new HashMap<>();

        for (int i = 0; i < s.length(); i++) {
            map.put(s.charAt(i), i);
        }
        for (int i = 0; i < t.length(); i++) {
            chars[i] = t.charAt(i);
        }

        Arrays.sort(chars, Comparator.comparing(x -> map.getOrDefault(x, 27)));
        char[] charArray = new char[t.length()];
        for (int i = 0; i < t.length(); i++) {
            charArray[i] = chars[i];
        }

        return new String(charArray);
    }
}
